expected scalar type Long but found Float

Pytorch tensor と numpy ndarray の変換 - Pythonいぬ (hatenablog.com) 主要还是 array 转 tensor的时候要确定类型 x = ...